java/Graph Editor/src/controllers/actions/DeleteNode.java

   1 package controllers.actions;
   2
   3 import javax.swing.undo.AbstractUndoableEdit;
   4
   5 import models.GraphEdge;
   6 import models.GraphVertex;
   7 import controllers.*;
   8
   9 public class DeleteNode extends AbstractUndoableEdit {
  10         private static final long serialVersionUID = -7929100929760152271L;
  11         private GraphVertex o;
  12         private SelectionController selectionController;
  13         private GraphEdge[] affectedEdges;
  14
  15         public DeleteNode(SelectionController selectionController)
  16         {
  17                 this.selectionController = selectionController;
  18                 this.o = selectionController.getSelectedVertex();
  19                 redoAction();
  20         }
  21
  22         private void removeAffectedEdges()
  23         {
  24                 selectionController.getModel().removeEdge(affectedEdges);
  25         }
  26
  27         private void addAffectedEdges()
  28         {
  29                 selectionController.getModel().addEdge(affectedEdges);
  30         }
  31
  32         private void getAffectedEdges(){
  33                 GraphEdge[] arr = selectionController.getModel().getEdges();
  34                 int count = 0;
  35                 for(int i=0; i < arr.length; i++)
  36                         if(arr[i].EdgeOf(o))
  37                                 count++;
  38                 count = 0;
  39                 affectedEdges = new GraphEdge[count];
  40                 for(int i=0; i < arr.length; i++)
  41                         if(arr[i].EdgeOf(o))
  42                                 affectedEdges[count++] = arr[i];
  43         }
  44
  45         private void redoAction()
  46         {
  47                 getAffectedEdges();
  48                 removeAffectedEdges();
  49                 selectionController.removeSelected();
  50                 selectionController.getModel().removeGraph(o);
  51         }
  52
  53         public void redo()
  54         {
  55                 super.redo();
  56                 redoAction();
  57         }
  58
  59         public void undo()
  60         {
  61                 super.undo();
  62                 addAffectedEdges();
  63                 selectionController.getModel().addGraph(o);
  64                 selectionController.setSelected(o);
  65         }
  66         public boolean isSignificant(){
  67                 return true;
  68         }
  69
  70         public String getPresentationName(){
  71                 return "Vertex verwijderen.";
  72         }
  73 }